import pandas as pd

# 替换为您的Excel文件路径
path_to_excel1 = 'summy_result.xlsx'
path_to_excel2 = 'extracted_unique_texts.xlsx'

# 读取Excel文件
df1 = pd.read_excel(path_to_excel1, header=None)
df2 = pd.read_excel(path_to_excel2, header=None)

# 初始化需要删除的列的索引集合
columns_to_drop = set()

# 遍历df1的每一列
for col_index in range(df1.shape[1]):
    # 检查第2行的数值是否小于10且第3行的数值是否等于0
    # 注意：iloc中索引从0开始，因此索引1和2分别对应第2行和第3行
    if df1.iloc[1, col_index] < 8 and df1.iloc[2, col_index] == 0:
        columns_to_drop.add(col_index)

# 删除df1中的对应列
df1.drop(columns=columns_to_drop, inplace=True)
df2.drop(columns=columns_to_drop, inplace=True)

# 保存更新后的表格到新的Excel文件
df1.to_excel('summy_result.xlsx', index=False)
df2.to_excel('train_data.xlsx', index=True)


path_to_excel = 'train_data.xlsx'

# 读取Excel文件，指定第一行为列名
df = pd.read_excel(path_to_excel, header= 1)
df3 = pd.read_excel('data2(800ture_label).xlsx') 
column_to_insert = df3.iloc[:, 3]
df.insert(len(df.columns), df3.columns[3], column_to_insert, True)




# 保存到Excel，不包含索引
df.to_excel('train_data.xlsx', index=False)

